package com.sprouting.ops.config;

import com.baomidou.mybatisplus.core.injector.AbstractMethod;
import com.baomidou.mybatisplus.core.injector.DefaultSqlInjector;
import com.baomidou.mybatisplus.core.metadata.TableInfo;
import com.baomidou.mybatisplus.extension.injector.methods.InsertBatchSomeColumn;
import com.baomidou.mybatisplus.extension.plugins.MybatisPlusInterceptor;
import com.baomidou.mybatisplus.extension.plugins.inner.PaginationInnerInterceptor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.List;

/**
 * MybatisPlus配置
 *
 * @author ：LX（长沙麓谷）
 * @date 创建时间： 2023/3/16 20:24
 */
@Configuration
@Slf4j
public class MybatisPlusConfig {

    /**
     * 配置mybatis-plus插件模块
     * @return
     */
    @Bean
    public MybatisPlusInterceptor mybatisPlusInterceptor() {
        MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
        // 分页拦截器
        interceptor.addInnerInterceptor(new PaginationInnerInterceptor());
        // interceptor.addInnerInterceptor(new PaginationInnerInterceptor(DbType.SQLITE));
        return interceptor;
    }


    /**
     * mp 支持批量插入，支持mysql、sqlite
     * @return
     */
    @Bean
    public DefaultSqlInjector mybatisPlusInsertBatchInjector(){
        return new DefaultSqlInjector() {
            /**
             * 批量插入支持
             * 原理就是在启动的时候就开始生成相关的sql语句，并不是在执行的时候才生成
             * @param mapperClass mapper文件
             * @param tableInfo 表信息
             * @return
             */
            @Override
            public List<AbstractMethod> getMethodList(Class<?> mapperClass, TableInfo tableInfo) {
                // 调用之前的注入方法，防止之前的失效
                List<AbstractMethod> methodList = super.getMethodList(mapperClass, tableInfo);
                // 增加批量支持,注意，这个默认的工具只支持 mysql 和部分数据库
                methodList.add(new InsertBatchSomeColumn());
                log.info("生成批量插入语句~~~~~~~~~~~~~~~~{}", tableInfo.getTableName());
                return methodList;
            }
        };
    }
}
